project(test_conv)

set(CUDA_NVCC_FLAGS -res-usage -shared -std c++14 -use_fast_math -g)

set(CUDA_SRC common.cu conv.cu gemm.cu im2col.cu test_conv.cu cudnn_conv.cu)

cuda_add_library(${PROJECT_NAME} SHARED ${CUDA_SRC})

install(TARGETS ${PROJECT_NAME} DESTINATION ${CMAKE_INSTALL_PREFIX}/lib)